from langchain.prompts import PromptTemplate
from langchain.llms import OpenAI

llm = OpenAI(temperature=0.9)
prompt = PromptTemplate(
    input_variables=["product"],
    template="What is a good name for a company that makes {product}?"
)

# 现在可以创建一个非常简单的链: 它接受用户输入，用它格式化提示符，然后将它发送到 LLM
from langchain.chains import LLMChain
chain = LLMChain(llm=llm, prompt=prompt)

# 运行该链，只指定产品
chain.run("colorful socks")